#pragma once

#include <iostream>
#include <queue>
#include <pthread.h>

const int gmaxcap = 500;
template <class T>
class BlockQueue
{
public:
    BlockQueue(const int &maxcap = gmaxcap)
        : _maxcap(maxcap)
    {
        pthread_mutex_init(&_mutex, nullptr);
        pthread_cond_init(&_pcond, nullptr);
        pthread_cond_init(&_ccond, nullptr);
    }

    void push(const T &in)
    {
        pthread_mutex_lock(&_mutex);
        // 1.判断
        // 这里必须要while，防止判断失败或者多个线程进行
        while (is_full())
        {
            //  第二个参数必须是我们正在使用的互斥锁
            //  1、调用该函数时，会以原子性的方式，将锁释放，并将自己挂起
            //  2、被唤醒返回的时候，会自动的重新获取锁
            pthread_cond_wait(&_pcond, &_mutex);
        }
        // 2.队列没满，生产者生产，入数据
        _q.push(in);
        // 3.消费者来消费，唤醒消费者
        pthread_cond_signal(&_ccond);   // 可以放在临界区内部，也可以放在外部
        pthread_mutex_unlock(&_mutex);
    }

    void pop(T *out)
    {
        pthread_mutex_lock(&_mutex);
        // 1.判断
        while (is_empty())
        {
            pthread_cond_wait(&_ccond, &_mutex);
        }
        // 2.队列不为空，消费者消费，出数据
        *out = _q.front();
        _q.pop();
        // 3.生产者来生产，唤醒生产者
        pthread_cond_signal(&_pcond);
        pthread_mutex_unlock(&_mutex);
    }

    ~BlockQueue()
    {
        pthread_mutex_destroy(&_mutex);
        pthread_cond_destroy(&_pcond);
        pthread_cond_destroy(&_ccond);
    }

private:
    bool is_empty()
    {
        return _q.empty();
    }
    bool is_full()
    {
        return _q.size() == _maxcap;
    }

private:
    std::queue<T> _q;
    int _maxcap; // 队列中元素的上限
    pthread_mutex_t _mutex;
    pthread_cond_t _pcond; // 生产者对应的条件变量
    pthread_cond_t _ccond; // 消费者对应的条件变量
};